import pytest
from core.base_page import BasePage  # 修正导入路径
from core.db_client import DBClient


@pytest.fixture(scope='function')
def db_client():
    """数据库连接fixture（类级别，每个测试类共享）"""
    db = DBClient()
    db.connect()
    yield db    # 传递数据库连接给测试用例
    db.close()  # 后置操作：关闭连接


@pytest.fixture(scope="function")
def init_browser():
    """初始化浏览器（类级别，每个测试类执行一次）"""
    base_page = BasePage()
    driver = base_page.driver
    yield driver  # 传递driver给测试用例
    base_page.quit()  # 后置操作：关闭浏览器

